python list comprehension multiple if else